from pydantic import BaseModel
from typing import List, Dict, Any

class NodeExecutionRequest(BaseModel):
    node_ids: List[str]
    workflow_id: int

class NodeExecutionResponse(BaseModel):
    success: bool
    message: str
    results: Dict[str, Any] 

class ExecuteNodesRequest(BaseModel):
    workflow_id: int
    node_ids: List[str]
    nodes: Dict[str, Any] = None 